package com.mybatisplus.mp.ifelse2;

import java.util.HashMap;
import java.util.Map;

public class DataUtils {
    private static Map<String, Class> ALL_DATA_SOURCE;
   static  {
       ALL_DATA_SOURCE = new HashMap<>();
    }
    /**
     * 根据策略名获取 Class
     * @param dsType
     * @return
     */
    public static Class getDs(String dsType) {
        return ALL_DATA_SOURCE.get(dsType);
    }

    /**
     * 策略名为 key,Class 为 value
     */
    public static void putClass(String dsType,Class clazz){
        ALL_DATA_SOURCE.put(dsType,clazz);
    }

}
